"""
ASGI config for devops_backend project.

It exposes the ASGI callable as a module-level variable named ``application``.

For more information on this file, see
https://docs.djangoproject.com/en/5.2/howto/deployment/asgi/
"""

import os
from django.core.asgi import get_asgi_application
from channels.routing import ProtocolTypeRouter, URLRouter
from channels.auth import AuthMiddlewareStack
from django.urls import path
from ssh_terminal.consumers import SSHTerminalConsumer

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'devops_backend.settings')

# Django ASGI应用
django_asgi_app = get_asgi_application()

# WebSocket路由
websocket_urlpatterns = [
    path('ws/ssh/<int:host_id>/', SSHTerminalConsumer.as_asgi()),
]

# ASGI应用
application = ProtocolTypeRouter({
    "http": django_asgi_app,
    "websocket": AuthMiddlewareStack(
        URLRouter(websocket_urlpatterns)
    ),
})
